import { BrainParameters } from "../Policies/BrainParameters";
import { DemonstrationMetaData } from "./DemonstrationMetaData";

export class DemonstrationSummary {
    public metaData: DemonstrationMetaData = null!;
    public brainParameters: BrainParameters = null!;
    public observationSummaries: ObservationSummary[] = null!;

    public initialize(brainParams: BrainParameters, demonstrationMetaData: DemonstrationMetaData, obsSummaries: ObservationSummary[]): void {
        this.brainParameters = brainParams;
        this.metaData = demonstrationMetaData;
        this.observationSummaries = obsSummaries;
    }
}

export class ObservationSummary {
    public shape: number[] = null!;
}